/*
 * Copyright The OpenTelemetry Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { toBinary, fromBinary, fromJsonString } from '@bufbuild/protobuf';
import { ISerializer } from '../../i-serializer';
import { createExportMetricsServiceRequest } from '../internal';
import { ResourceMetrics } from '@opentelemetry/sdk-metrics';
import { IExportMetricsServiceResponse } from '../export-response';
import {
  ExportMetricsServiceRequestSchema,
  ExportMetricsServiceResponseSchema,
} from '../../generated/opentelemetry/proto/collector/metrics/v1/metrics_service_pb';
import { PROTOBUF_JSON_ENCODER } from '../../common/utils';

export const ProtobufMetricsSerializer: ISerializer<
  ResourceMetrics,
  IExportMetricsServiceResponse
> = {
  serializeRequest: (arg: ResourceMetrics) => {
    const request = createExportMetricsServiceRequest(
      [arg],
      PROTOBUF_JSON_ENCODER
    );
    const message = fromJsonString(
      ExportMetricsServiceRequestSchema,
      JSON.stringify(request)
    );
    return toBinary(ExportMetricsServiceRequestSchema, message);
  },
  deserializeResponse: (arg: Uint8Array) => {
    const response = fromBinary(ExportMetricsServiceResponseSchema, arg);
    return {
      partialSuccess: response.partialSuccess
        ? {
            rejectedDataPoints: Number(
              response.partialSuccess.rejectedDataPoints
            ),
            errorMessage: response.partialSuccess.errorMessage,
          }
        : undefined,
    };
  },
};
